Faster numpy array indexing when using condition (numpy.where)